package com.ym.component;

import com.ym.utils.RequestUtils;
import feign.RequestInterceptor;
import feign.RequestTemplate;
import org.springframework.stereotype.Component;

/**
 * feign拦截器, 在feign请求发出之前，加入一些操作
 */
@Component
public class FeignInterceptor implements RequestInterceptor {
    // 为 Feign 的 RPC 调用 添加请求头Same-Token
    @Override
    public void apply(RequestTemplate requestTemplate) {
        String token = RequestUtils.getCurrentToken();
        if (token != null) {
            requestTemplate.header("sa-token", token);
        }

        String sameToken = RequestUtils.getSameToken();
        if (sameToken != null) {
            requestTemplate.header("SA-SAME-TOKEN", sameToken);
        }
    }
}

